package org.example;

import java.util.PriorityQueue;
import java.util.Random;

public class Test1 {
    //leetcode 215 数组中的第k个最大元素 https://leetcode.cn/problems/kth-largest-element-in-an-array/description/

    /**
     * 通过随机选取一个数作为基准值，将数组中的元素分为小于，等于和大于这个基准值的部分，然后判断这个基准值是第几大再决定去那个部分去找
     * @param nums
     * @param k
     * @return
     */
    public int findKthLargest(int[] nums, int k) {
        return qsort(nums,0,nums.length-1,k);
    }

    private int qsort(int[] nums, int l, int r, int k) {
        int key = nums[new Random().nextInt(r - l + 1) + l];
        int left = l-1, right = r + 1, i = l;
        while(i < right) {
            if(nums[i] < key) swap(nums,++left,i++);
            else if(nums[i] == key) i++;
            else swap(nums,--right,i);
        }
        //c表示大于key的部分，b表示等于key的部分，剩下的部分就是小于key的部分
        int c = r - right + 1;
        int b = right - left - 1;
        if(c >= k) return qsort(nums,right,r,k);
        else if(b + c >= k) return key;
        else return qsort(nums,l,left,k - b - c);
    }

    private void swap(int[] nums, int x, int y) {
        int tmp = nums[x];
        nums[x] = nums[y];
        nums[y] = tmp;
    }

    public static void main(String[] args) {
        Test1 test1 = new Test1();
        int[] arr = {3,2,3,1,2,4,5,5,6};
        int ret = test1.findKthLargest(arr,20);
        System.out.println(ret);
    }
}
